/*
 *
 * Copyright (c) 2020-2022, Java知识图谱 (http://www.altitude.xin).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package xin.altitude.cms.common.util;

import org.redisson.api.RLock;
import xin.altitude.cms.common.entity.LockMeta;

import java.util.Objects;
import java.util.Optional;
import java.util.function.Function;

/**
 * 锁相关工具类
 *
 * @author <a href="http://www.altitude.xin" target="_blank">Java知识图谱</a>
 * @author <a href="https://gitee.com/decsa/ucode-cms-vue" target="_blank">UCode CMS</a>
 * @author <a href="https://space.bilibili.com/1936685014" target="_blank">B站视频</a>
 **/
public class LockUtils {
    private LockUtils() {
    }


    /**
     * 通过封装减少try-catch冗余代码
     *
     * @param meta   {@link LockMeta}锁元数据子类实例
     * @param u      具体业务相关的参数
     * @param action 回调方法
     * @param <T>    {@link LockMeta}锁元数据子类
     * @param <R>    返回值类型
     */
    public static <T extends LockMeta, U, R> R tryLock(T meta, U u, Function<U, R> action) {
        RLock lock = Objects.requireNonNull(meta.getLock());
        try {
            if (lock.tryLock(meta.getWaitTime(), meta.getLeaseTime(), meta.getTimeUnit())) {
                // 回调被锁业务逻辑
                return action.apply(u);
            }
            return null;
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } finally {
            Optional.of(lock).ifPresent(RLock::unlock);
        }
    }
}
